import pandas as pd
from sqlalchemy import create_engine

# 读取Excel文件
excel_file = 'D://cxdownload//职工信息.xlsx'  # 替换为你的Excel文件路径
df = pd.read_excel(excel_file)

# 创建数据库连接
db_type = 'mysql'  # 数据库类型
username = 'root'  # 数据库用户名
password = '521314'  # 数据库密码
host = '127.0.0.1'  # 数据库主机地址
port = '3306'  # MySQL默认端口是3306，请确保端口号正确
database = 'factory'  # 数据库名
table_name = 'salary'  # 表名

# 创建数据库引擎
engine = create_engine(f"{db_type}+pymysql://{username}:{password}@{host}:{port}/{database}")

# 指定Excel列对应数据库表列的映射关系
columns_mapping = {
    '职工号': 'zgh',
    '姓名': 'zg_name',
    '日期': 'ffrq',
    '工资': 'gz',
    # '性别': 'xb',
    # '出生日期': 'csrq',
    # '政治面貌': 'zzmm',
    # '参加工作': 'gzsj',
    # '': 'bmh',
    # '部门名': 'bm-name',
    # 添加更多的映射关系
}

# 重命名DataFrame中的列以匹配数据库表的列名
df.rename(columns=columns_mapping, inplace=True)

# 将数据导入数据库，只导入指定的列
df.to_sql(table_name, con=engine, if_exists='append', index=False)